package com.atguigu.digest;

import org.junit.Test;

import java.io.ByteArrayOutputStream;
import java.io.FileInputStream;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;

/**
 *
 * @author : 马伟奇
 * CreateTime: 2020-05-05
 * Description:
 * 消息摘要算法，为了防止篡改
 * 常见的加密算法：md5，sha1 ，sha256，sha512
 */

public class DigestDemo   {



    /**
     * 获取文件的数字摘要
     *
     * @param filePath  表示文件的路径
     * @param algorithm 表示算法
     * @return
     */
    private static String getDigestFile(String filePath, String algorithm) throws Exception {
        // 通过io流的方式读取文件
        FileInputStream fis = new FileInputStream(filePath);
        // 定义长度
        int len;
        // 定义一个字节数组
        byte[] buffer = new byte[1024];
        // 边读边写
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        while (((len = fis.read(buffer)) != -1)) {
            baos.write(buffer, 0, len);
        }
        // 创建消息摘要对象
        MessageDigest digest = MessageDigest.getInstance(algorithm);
        // 执行消息摘要算法
        byte[] digest1 = digest.digest(baos.toByteArray());

        return toHex(digest1);
    }

    /**
     * 获取数字摘要
     *
     * @param input     原文
     * @param algorithm 算法
     * @return 16进制的密文
     * @throws NoSuchAlgorithmException 算法名 错误
     */
    private static String getDigest(String input, String algorithm) throws NoSuchAlgorithmException {
        // 创建消息摘要对象
        MessageDigest digest = MessageDigest.getInstance(algorithm);
        // 执行消息摘要算法
        byte[] digest1 = digest.digest(input.getBytes());

        return toHex(digest1);
    }


    /**
     * 加密 数据转成16进制
     */
    private static String toHex(byte[] digest) {
        StringBuilder sb = new StringBuilder();
        // 对密文进行迭代
        for (byte b : digest) {
            // 把密文转换成16进制
            String s = Integer.toHexString(b & 0xff);
            // 判断如果密文的长度是1，需要在高位进行补0
            if (s.length() == 1) {
                s = "0" + s;
            }
            sb.append(s);
//            System.out.print(s);
        }
        // 使用base64进行转码
        return sb.toString();
//        System.out.println(sb.toString());
    }


    private String input = "艹";


    /**
     * md5加密
     * 16位 小写	ffeabe769b5d5ebd
     * 16位 大写	FFEABE769B5D5EBD
     * 32位 小写	748284b6ffeabe769b5d5ebd40d7c683
     * 32位 大写	748284B6FFEABE769B5D5EBD40D7C683
     *
     *    md5===748284b6ffeabe769b5d5ebd40d7c683
     * */
    @Test
    public void testMd5() throws NoSuchAlgorithmException {

        // 算法

        String algorithm = "MD5";
        String md5 = getDigest(input, algorithm);

        System.out.println("md5===" + md5);


    }

    /**
     * sha1===35746d663e247f93c91e754219f8e8f862139611
     * */
    @Test
    public  void testSha_1 () throws NoSuchAlgorithmException {


        String sha1 = getDigest(input, "SHA-1");
        System.out.println("sha1==="+sha1);
    }


    /**
     * sha256===34e1840d9f2e3382714a16d51afa83b00589697c9cf72d39f041636efebfa78b
     * */
    @Test
    public void testSha_256 () throws NoSuchAlgorithmException {

        String sha256 = getDigest(input, "SHA-256");
        System.out.println("sha256==="+sha256);
    }

    /**
     * sha512===6b4316fcd573c806c3532e1c8f5a3809e9d0e4e0271413c9905a8f21a3db5050d260131e835ae97fb84f8eb81d81ae1e9415b2b2524700a8e8a8783dfadaff5d
     * */
    @Test
    public void testSha_512 () throws NoSuchAlgorithmException {
        String sha512 = getDigest(input, "SHA-512");
        System.out.println("sha512==="+sha512);

    }


    public static void main(String[] args) throws Exception {

        String sha1 = getDigestFile("11.zip", "SHA-1");
        System.out.println(sha1);
        //dc3743e7bb409de25d1df26656cad1e15c206af17751f0d8ae5898c1325efac0c69918228649696f32126e4e32c99a73b422ef96b61cef75be384a4d0aef9c9f
        String sha512 = getDigestFile("11.zip", "SHA-512");
        System.out.println(sha512);


    }


}
